#pragma once

#include <opencv2/opencv.hpp>

#include "light_detect.h"

// 装甲板
struct ArmorAnnotation {
  cv::Rect bbox; // 装甲板 Boundbox
  int class_id;  // 装甲板种类
  bool color; // 颜色
};

// 解析 xml 文件，提取装甲板信息
std::vector<ArmorAnnotation> parseXmlFile(const std::string &xml_path);

// 处理图像
bool findAndWarpArmor(cv::Mat& src, cv::Mat& dst, bool color);